# coding=utf-8
# 代码文件：chapter2/ch2.5.py
# 导入Selenium模块
from selenium import webdriver

# 创建 Firefox浏览器使用的Webdriver对象
driver = webdriver.Firefox()

url = 'http://q.stock.sohu.com/cn/{0}/lshq.shtml'
# 传递参数获得最终URL网址
strURL = url.format('600519')  # 股票代码
print("请求的URL：", strURL)

# 发送请求
driver.get(strURL)

# 通过id找到表格标签对象
tableElement = driver.find_element_by_id('BIZ_hq_historySearch')

# 通过标签名找出表格中找出所有的tr元素
trList = tableElement.find_elements_by_tag_name('tr')

# 保存数据列表
datas = []
for index, tr in enumerate(trList):
    # 跳过表格前四行数据
    if index < 4:
        continue
    # 找tr下面所有的元素
    tds = tr.find_elements_by_tag_name('td')

    #  保存一行数据的字典对象
    row = dict()
    # 日期
    row['Date'] = tds[0].text
    # 开盘价
    row['Open'] = float(tds[1].text.replace(',', ''))
    # 收盘价
    row['Close'] = float(tds[2].text.replace(',', ''))
    # 最高价
    row['High'] = float(tds[5].text.replace(',', ''))
    # 最低价
    row['Low'] = float(tds[6].text.replace(',', ''))
    #  	成交量
    row['Volume'] = int(tds[7].text.replace(',', ''))
    datas.append(row)
# 测试一下爬取回来的数据
print(datas)

# 退出浏览器
driver.quit()
